Backdrop

백준 온라인 저지 ▸ 24511

queuestack
III

문제

한가롭게 방학에 놀고 있던 도현이는 갑자기 재밌는 자료구조를 생각해냈다. 그 자료구조의 이름은 queuestack이다.

queuestack의 구조는 다음과 같다. 11번, 22번, ... , NN번의 자료구조(queue 혹은 stack)가 나열되어있으며, 각각의 자료구조에는 한 개의 원소가 들어있다.

queuestack의 작동은 다음과 같다.

  • x0x_0을 입력받는다.
  • x0x_011번 자료구조에 삽입한 뒤 11번 자료구조에서 원소를 pop한다. 그때 pop된 원소를 x1x_1이라 한다.
  • x1x_122번 자료구조에 삽입한 뒤 22번 자료구조에서 원소를 pop한다. 그때 pop된 원소를 x2x_2이라 한다.
  • ...
  • xN1x_{N-1}NN번 자료구조에 삽입한 뒤 NN번 자료구조에서 원소를 pop한다. 그때 pop된 원소를 xNx_N이라 한다.
  • xNx_N을 리턴한다.

도현이는 길이 MM의 수열 CC를 가져와서 수열의 원소를 앞에서부터 차례대로 queuestack에 삽입할 것이다. 이전에 삽입한 결과는 남아 있다. (예제 11 참고)

queuestack에 넣을 원소들이 주어졌을 때, 해당 원소를 넣은 리턴값을 출력하는 프로그램을 작성해보자.

입력

첫째 줄에 queuestack을 구성하는 자료구조의 개수 NN이 주어진다. (1N1000001 \leq N \leq 100\,000)

둘째 줄에 길이 NN의 수열 AA가 주어진다. ii번 자료구조가 큐라면 Ai=0A_i = 0, 스택이라면 Ai=1A_i = 1이다.

셋째 줄에 길이 NN의 수열 BB가 주어진다. BiB_iii번 자료구조에 들어 있는 원소이다. (1Bi10000000001 \leq B_i \leq 1\,000\,000\,000)

넷째 줄에 삽입할 수열의 길이 MM이 주어진다. (1M1000001 \leq M \leq 100\,000)

다섯째 줄에 queuestack에 삽입할 원소를 담고 있는 길이 MM의 수열 CC가 주어진다. (1Ci10000000001 \leq C_i \leq 1\,000\,000\,000)

입력으로 주어지는 모든 수는 정수이다.

출력

수열 CC의 원소를 차례대로 queuestack에 삽입했을 때의 리턴값을 공백으로 구분하여 출력한다.

예제 입력 1

4
0 1 1 0
1 2 3 4
3
2 4 7

예제 출력 1

4 1 2

각 상태에 대한 큐스택 내부를 표현하면 다음과 같다.

  • 초기 상태 : [1,2,3,4][1, 2, 3, 4]
  • 첫 번째 원소 삽입 : [2,2,3,1][2, 2, 3, 1]
  • 두 번째 원소 삽입 : [4,2,3,2][4, 2, 3, 2]
  • 세 번째 원소 삽입 : [7,2,3,4][7, 2, 3, 4]

예제 입력 2

5
1 1 1 1 1
1 2 3 4 5
3
1 3 5

예제 출력 2

1 3 5

풀이

코드

from collections import deque
 
 
n = int(input())
a = list(map(int, input().split()))
b = list(map(int, input().split()))
m = int(input())
c = list(map(int, input().split()))
 
d = deque(c)
for ai, bi in zip(a, b):
    if ai == 0:
        d.appendleft(bi)
 
print(*(list(d)[:m]))